from fastapi import FastAPI
from fastapi.websockets import WebSocket
from fastapi.testclient import TestClient
from main import app

def test_websocket():
    client = TestClient(app)
    with client.websocket_connect("/ws/upload") as websocket:
        websocket.
        data = websocket.receive_json()
        assert data == {"msg": "Hello WebSocket"}